-
Notifications
You must be signed in to change notification settings - Fork 2.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added a Dockerfile and instructions to build it #163
base: main
Are you sure you want to change the base?
Conversation
sergiobd
commented
Sep 5, 2023
- Based FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
- "hacked" some cmake files in order for things to build correctly. Mainly:
- OpenCV version 4.5 -> 4.2
- Tweaks to compile Embree library
[BTW: I am simply a user and not a maintainer of this repo] Although this looks very good, I could not run it, see the extract log of the docker build |
It seems that CUDA is not available during build. Did you set Your host's
After doing this, restart docker with |
Actually this was due to a missing variable, as explained in pytorch/extension-cpp#71
to the Dockerfile. |
|
||
# Create a Conda environment and activate it | ||
WORKDIR /workspace/gaussian-splatting | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
add somewhere ( for CUDA >11.0 )
ARG TORCH_CUDA_ARCH_LIST="3.5;5.0;6.0;6.1;7.0;7.5;8.0;8.6+PTX"
Tested, works great with cuda 12 |
This step still might not be enough with newer docker versions. In addition to updating the default runtime build your image with:
Reference: https://stackoverflow.com/questions/59691207/docker-build-with-nvidia-runtime/75629058#75629058 |